#!/usr/bin/env bash
BASE_DIR=${BASE_DIR:-"/public/home/jianghaitenggroup/jianglab/WorkSpace/Qwen25_VL_FT"}

CKPT_DIR=${CKPT_DIR:-"${BASE_DIR}/saved/train/best"}
SYSTEM_PROMPT=${SYSTEM_PROMPT:-"${BASE_DIR}/system_prompt.md"}
SUBJECTS_JSON=${SUBJECTS_JSON:-"${BASE_DIR}/saved/data/subjects.json"}
IMAGE_ROOT=${IMAGE_ROOT:-"/public/home/jianghaitenggroup/jianglab/WorkSpace/MASS-SS3_Wave_224_448_V2/"}
BATCH_SIZE=${BATCH_SIZE:-128}
OUTPUT_JSONL=${OUTPUT_JSONL:-"${BASE_DIR}/saved/test/results.jsonl"}
OUTPUT_XLSX=${OUTPUT_XLSX:-"${BASE_DIR}/saved/test/metrics.xlsx"}

cd "$BASE_DIR"

python ./eval.py \
    --ckpt_dir "$CKPT_DIR" \
    --system_prompt "$SYSTEM_PROMPT" \
    --subjects_json "$SUBJECTS_JSON" \
    --subjects_split "test" \
    --img_dir "$IMAGE_ROOT" \
    --batch_size "$BATCH_SIZE" \
    --output_jsonl "$OUTPUT_JSONL" \
    --output_xlsx "$OUTPUT_XLSX"
